#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
创建测试Excel文件
用于演示merge_excel_sheets.py脚本的功能
"""

import pandas as pd

def create_test_excel():
    """创建一个包含多个工作表的测试Excel文件"""
    
    print("📝 正在创建测试Excel文件...")
    
    # 创建第一个工作表（员工信息）
    sheet1_data = {
        '员工ID': ['E001', 'E002', 'E003'],
        '姓名': ['张三', '李四', '王五'],
        '部门': ['技术部', '销售部', '人事部'],
        '工资': [10000, 8000, 7000]
    }
    df1 = pd.DataFrame(sheet1_data)
    
    # 创建第二个工作表（更多员工）
    sheet2_data = {
        '员工ID': ['E004', 'E005'],
        '姓名': ['赵六', '钱七'],
        '部门': ['技术部', '财务部'],
        '工资': [12000, 9000]
    }
    df2 = pd.DataFrame(sheet2_data)
    
    # 创建第三个工作表（新入职员工）
    sheet3_data = {
        '员工ID': ['E006', 'E007', 'E008'],
        '姓名': ['孙八', '周九', '吴十'],
        '部门': ['销售部', '技术部', '人事部'],
        '工资': [7500, 11000, 6500]
    }
    df3 = pd.DataFrame(sheet3_data)
    
    # 保存到Excel文件
    file_name = 'test_employee_data.xlsx'
    with pd.ExcelWriter(file_name, engine='openpyxl') as writer:
        df1.to_excel(writer, sheet_name='第一季度', index=False)
        df2.to_excel(writer, sheet_name='第二季度', index=False)
        df3.to_excel(writer, sheet_name='第三季度', index=False)
    
    print(f"✅ 测试文件创建成功: {file_name}")
    print(f"\n文件包含:")
    print(f"  - 工作表1 '第一季度': {len(df1)} 行数据")
    print(f"  - 工作表2 '第二季度': {len(df2)} 行数据")
    print(f"  - 工作表3 '第三季度': {len(df3)} 行数据")
    print(f"  - 合并后预期: {len(df1) + len(df2) + len(df3)} 行数据")
    
    return file_name

if __name__ == "__main__":
    create_test_excel()
